package com.yunxi.module.system.mapper.oauth2;

import com.yunxi.framework.mybatis.core.mapper.BaseMapperX;
import com.yunxi.framework.mybatis.core.query.LambdaQueryWrapperX;
import com.yunxi.framework.tenant.core.aop.TenantIgnore;
import com.yunxi.module.system.model.oauth2.OAuth2AccessTokenDO;
import org.apache.ibatis.annotations.Mapper;

import java.util.List;

/**
 * <p>
 * OAuth2 的访问令牌 mapper
 * </p>
 *
 * @author lidy
 * @since 2024-05-08
 */
@Mapper
public interface OAuth2AccessTokenMapper extends BaseMapperX<OAuth2AccessTokenDO> {

    /**
     * 根据 accessToken 查询 OAuth2 访问令牌
     * 获取 token 的时候，需要忽略租户编号。原因是：一些场景下，可能不会传递 tenant-id 请求头，例如说文件上传、积木报表等等
     *
     * @param accessToken token
     * @return 访问令牌信息
     */
    @TenantIgnore
    default OAuth2AccessTokenDO selectByAccessToken(String accessToken) {
        return selectOne(OAuth2AccessTokenDO::getAccessToken, accessToken);
    }

    /**
     * 根据刷新令牌删除
     *
     * @param refreshToken 刷新令牌
     */
    default int deleteByRefreshToken(String refreshToken) {
        return delete(new LambdaQueryWrapperX<OAuth2AccessTokenDO>().eq(OAuth2AccessTokenDO::getRefreshToken, refreshToken));
    }

    /**
     * 查询访问令牌集合
     *
     * @param refreshToken 刷新令牌
     * @return 访问令牌集合
     */
    default List<OAuth2AccessTokenDO> selectListByRefreshToken(String refreshToken) {
        return selectList(OAuth2AccessTokenDO::getRefreshToken, refreshToken);
    }
}
